package com.shm.leetcode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * 733. 图像渲染
 * 有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。
 *
 * 给你一个坐标 (sr, sc) 表示图像渲染开始的像素值（行 ，列）和一个新的颜色值 newColor，让你重新上色这幅图像。
 *
 * 为了完成上色工作，从初始坐标开始，记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
 *
 * 最后返回经过上色渲染后的图像。
 *
 * 示例 1:
 *
 * 输入:
 * image = [[1,1,1],[1,1,0],[1,0,1]]
 * sr = 1, sc = 1, newColor = 2
 * 输出: [[2,2,2],[2,2,0],[2,0,1]]
 * 解析:
 * 在图像的正中间，(坐标(sr,sc)=(1,1)),
 * 在路径上所有符合条件的像素点的颜色都被更改成2。
 * 注意，右下角的像素没有更改为2，
 * 因为它不是在上下左右四个方向上与初始点相连的像素点。
 * 注意:
 *
 * image 和 image[0] 的长度在范围 [1, 50] 内。
 * 给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
 * image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。
 * @author SHM
 */
public class FloodFill {
    int[][] direction = {{1,0},{-1,0},{0,1},{0,-1}};

    /**
     * 方法一：广度优先搜索
     * 思路及算法
     *
     * 我们从给定的起点开始，进行广度优先搜索。每次搜索到一个方格时，如果其与初始位置的方格颜色相同，就将该方格加入队列，并将该方格的颜色更新，以防止重复入队。
     *
     * 注意：因为初始位置的颜色会被修改，所以我们需要保存初始位置的颜色，以便于之后的更新操作。
     * 复杂度分析
     *
     * 时间复杂度：O(n\times m)O(n×m)，其中 nn 和 mm 分别是二维数组的行数和列数。最坏情况下需要遍历所有的方格一次。
     * 空间复杂度：O(n\times m)O(n×m)，其中 nn 和 mm 分别是二维数组的行数和列数。主要为队列的开销。
     * 作者：LeetCode-Solution
     * 链接：https://leetcode-cn.com/problems/flood-fill/solution/tu-xiang-xuan-ran-by-leetcode-solution/
     * @param image
     * @param sr
     * @param sc
     * @param newColor
     * @return
     */
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        int currentColor = image[sr][sc];
        if(currentColor == newColor){
            return image;
        }
        int n = image.length,m = image[0].length;
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(new int[]{sr,sc});
        image[sr][sc] = newColor;
        while(!queue.isEmpty()){
            int[] cell = queue.poll();
            int x = cell[0], y = cell[1];
            for(int[] d:direction){
                int mx = x+d[0],my = y+d[1];
                if(mx>=0&&my>=0&&mx<n&&my<m&&image[mx][my]==currentColor){
                    queue.offer(new int[]{mx,my});
                    image[mx][my] = newColor;
                }
            }
        }
        return image;
    }

    /**
     * 方法二：深度优先搜索
     * 思路及算法
     *
     * 我们从给定的起点开始，进行深度优先搜索。每次搜索到一个方格时，如果其与初始位置的方格颜色相同，就将该方格的颜色更新，以防止重复搜索；如果不相同，则进行回溯。
     *
     * 注意：因为初始位置的颜色会被修改，所以我们需要保存初始位置的颜色，以便于之后的更新操作。
     *
     * 复杂度分析
     *
     * 时间复杂度：O(n\times m)O(n×m)，其中 nn 和 mm 分别是二维数组的行数和列数。最坏情况下需要遍历所有的方格一次。
     * 空间复杂度：O(n\times m)O(n×m)，其中 nn 和 mm 分别是二维数组的行数和列数。主要为栈空间的开销。
     * 作者：LeetCode-Solution
     * 链接：https://leetcode-cn.com/problems/flood-fill/solution/tu-xiang-xuan-ran-by-leetcode-solution/
     * @param image
     * @param sr
     * @param sc
     * @param newColor
     * @return
     */
    public int[][] floodFill_2(int[][] image, int sr, int sc, int newColor) {
        int currentColor = image[sr][sc];
        if(currentColor != newColor){
            dfs(image,sr,sc,currentColor,newColor);
        }
        return image;
    }

    public void dfs(int[][] image,int sr, int sc,int currentColor,int newColor){
        int n = image.length,m = image[0].length;
        image[sr][sc] = newColor;
        for(int[] d:direction){
            int mx = sr+d[0],my = sc+d[1];
            if(mx>=0&&my>=0&&mx<n&&my<m&&image[mx][my]==currentColor){
                dfs(image,mx,my,currentColor,newColor);
            }
        }
    }
}
